import os
import sys
sys.path.append(os.path.abspath('.'))
import jittor as jt
from jittor.models import vgg16, vgg16_bn, resnet50
import jittor.nn as nn


import numpy as np

# from models.vgg import vgg16
from models.resnet import resnet50


model = resnet50()#pretrained=True

a=model.state_dict()

for k in a.keys():
    print(k)

c=100